# 导入预测工具
import detect

# 运行预测程序：指定要预测的图片，预训练好的模型及是否显示图片
# detect.run(source=0, weights='yolov5s.pt', view_img=False,device='mps')

# detect.run(source='data/images/bus.jpg', weights='yolov5s.pt', view_img=True,device='mps')

detect.run(source='data/worker_safe/images/val/ppe_0000_jpg.rf.7b03fb2e890ac3f5c8e6577280145d0f.jpg',
           weights='best.pt', view_img=True,device='mps')